import winreg as reg
import os

def add_context_menu(menu_name,command,reg_root_key_path,reg_key_path,shortcut_key):
    # 打开名称父键
    key = reg.OpenKey(reg_root_key_path, reg_key_path, 0, reg.KEY_SET_VALUE)
    # 为key创建一个名称为menu_name的sub_key，并设置sub_key的值为menu_name加上快捷键，数据类型为REG_SZ字符串类型
    reg.SetValue(key, menu_name, reg.REG_SZ, menu_name + '(&{0})'.format(shortcut_key))

    # 打开刚刚创建的名为menu_name的sub_key
    sub_key = reg.OpenKey(key, menu_name, 0, reg.KEY_SET_VALUE)
    # 为sub_key添加名为'command'的子键，并设置其值为command + ' "%v"'，数据类型为REG_SZ字符串类型
    reg.SetValue(sub_key, 'command', reg.REG_SZ, command + ' "%v"')

    # 关闭sub_key和key
    reg.CloseKey(sub_key)
    reg.CloseKey(key)

def add_kzkt_cut_shortcut():
    # 右键菜单名
    menu_name = '一键剪辑"空中课堂"'
    # Chrome浏览器可执行文件的本地绝对路径
    cwd = os.getcwd()
    command = r'cmd.exe /c python '+ cwd + r'\kzkt-cut.py --video'
    # 注册表根键
    reg_root_key_path = reg.HKEY_CLASSES_ROOT
    # 注册表父键
    reg_key_path = r'*\\shell'
    # 快捷键
    shortcut_key = 'U'
    add_context_menu(menu_name, command, reg_root_key_path, reg_key_path, shortcut_key)

add_kzkt_cut_shortcut()